package com.sk.sk_frame_cloud.config;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

import java.util.Map;

/**
 * 动态数据源
 * 调用AbstractRoutingDataSource组件的addDefineDynamicDataSource（）方法，获取原来targetdatasources的map，
 * 并将新的数据源信息添加到map中，并替换targetdatasources中的map
 * 切换数据源时可以使用@DataSource(value = "数据源名称")，或者DynamicDataSourceContextHolder.setContextKey("数据源名称")
 * @Author  jiaok
 * @Date 2023-12-11 18:20
 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public class DynamicDataSourceConfig extends AbstractRoutingDataSource {

    /**
     * 备份所有数据源信息，
     */
    private  Map<Object, Object>   defineTargetDataSources;


    /**
     * 默认为 baseDB数据源  数据源动态选择策略
     * @return
     */
    @Override
    protected Object determineCurrentLookupKey() {
        return DynamicDataSourceHolder.getDynamicDataSourceKey();
    }
}
